package com.javaee.wordtree.entity;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;

import java.util.Date;

/**
 * 消息实体类
 */
@Data
@TableName("messages")
public class Message {

    /**
     * 消息ID
     */
    @TableId(value = "message_id", type = IdType.AUTO)
    private Integer messageId;

    /**
     * 聊天ID
     */
    private Integer chatId;

    /**
     * 消息文本
     */
    private String text;

    /**
     * 是否为用户发送的消息
     */
    private Boolean isUser;
    /**
     * 消息发送时间
     */
    private Date createdTime;

    /**
     * 是否基于RAG回答
     */
    private Boolean basedOnRag;

    /**
     * 检索到的RAG信息数量
     */
    private Integer ragCount;

    /**
     * 过滤后有效的RAG信息数量
     */
    private Integer validRagCount;

    /**
     * 平均相似度
     */
    private Double averageSimilarity;

    /**
     * 最高相似度
     */
    private Double maxSimilarity;

    /**
     * 最低相似度
     */
    private Double minSimilarity;

    /**
     * 相似度阈值
     */
    private Double similarityThreshold;

    /**
     * RAG结果详情（JSON格式存储）
     */
    private String ragResults;
}
